import org.apache.spark.sql.SparkSession

val spark = SparkSession.builder
val salesDF = spark.read.format("csv")
  .option("header", "true")
  .option("inferSchema", "true")
  .load("sales_data.csv")


val selectedDF = salesDF
  .select($"OrderID", $"Product", $"Quantity".as("UnitsSold"))
  
selectedDF.show()

